import unittest
import narcissus_numbers


class MyTestCase(unittest.TestCase):
    def test_narcissus(self):
        self.assertSequenceEqual([153, 370, 371, 407], list(narcissus_numbers.generate_narcissus()))
        self.assertSequenceEqual([153, 370, 371, 407], narcissus_numbers.narcissus_list())


if __name__ == '__main__':
    unittest.main()
